Calling functions in module by choice in main python program